home *** CD-ROM | disk | FTP | other *** search
/ Personal Computer World 2008 February / PCWFEB08.iso / Software / Freeware / Miro 1.0 / Miro_Installer.exe / xulrunner / python / BitTorrent / fakeopen.py < prev    next >
Encoding:
Python Source  |  2007-11-12  |  2.2 KB  |  90 lines
  1. # Written by Bram Cohen
  2. # see LICENSE.txt for license information
  3.  
  4. from string import join
  5.  
  6. class FakeHandle:
  7.     def __init__(self, name, fakeopen):
  8.         self.name = name
  9.         self.fakeopen = fakeopen
  10.         self.pos = 0
  11.     
  12.     def flush(self):
  13.         pass
  14.     
  15.     def close(self):
  16.         pass
  17.     
  18.     def seek(self, pos):
  19.         self.pos = pos
  20.     
  21.     def read(self, amount = None):
  22.         old = self.pos
  23.         f = self.fakeopen.files[self.name]
  24.         if self.pos >= len(f):
  25.             return ''
  26.         if amount is None:
  27.             self.pos = len(f)
  28.             return join(f[old:], '')
  29.         else:
  30.             self.pos = min(len(f), old + amount)
  31.             return join(f[old:self.pos], '')
  32.     
  33.     def write(self, s):
  34.         f = self.fakeopen.files[self.name]
  35.         while len(f) < self.pos:
  36.             f.append(chr(0))
  37.         self.fakeopen.files[self.name][self.pos : self.pos + len(s)] = list(s)
  38.         self.pos += len(s)
  39.  
  40. class FakeOpen:
  41.     def __init__(self, initial = {}):
  42.         self.files = {}
  43.         for key, value in initial.items():
  44.             self.files[key] = list(value)
  45.     
  46.     def open(self, filename, mode):
  47.         """currently treats everything as rw - doesn't support append"""
  48.         self.files.setdefault(filename, [])
  49.         return FakeHandle(filename, self)
  50.  
  51.     def exists(self, file):
  52.         return self.files.has_key(file)
  53.  
  54.     def getsize(self, file):
  55.         return len(self.files[file])
  56.  
  57. def test_normal():
  58.     f = FakeOpen({'f1': 'abcde'})
  59.     assert f.exists('f1')
  60.     assert not f.exists('f2')
  61.     assert f.getsize('f1') == 5
  62.     h = f.open('f1', 'rw')
  63.     assert h.read(3) == 'abc'
  64.     assert h.read(1) == 'd'
  65.     assert h.read() == 'e'
  66.     assert h.read(2) == ''
  67.     h.write('fpq')
  68.     h.seek(4)
  69.     assert h.read(2) == 'ef'
  70.     h.write('ghij')
  71.     h.seek(0)
  72.     assert h.read() == 'abcdefghij'
  73.     h.seek(2)
  74.     h.write('p')
  75.     h.write('q')
  76.     assert h.read(1) == 'e'
  77.     h.seek(1)
  78.     assert h.read(5) == 'bpqef'
  79.  
  80.     h2 = f.open('f2', 'rw')
  81.     assert h2.read() == ''
  82.     h2.write('mnop')
  83.     h2.seek(1)
  84.     assert h2.read() == 'nop'
  85.     
  86.     assert f.exists('f1')
  87.     assert f.exists('f2')
  88.     assert f.getsize('f1') == 10
  89.     assert f.getsize('f2') == 4
  90.